package io.gatling.extensions.kafka

import io.gatling.extensions.kafka.builder.{KafkaConsumerActionBuilder, KafkaProductActionBuilder}
import io.gatling.extensions.kafka.protocol.{KafkaConsumerProtocol, KafkaConsumerProtocolBuilder, KafkaProducerProtocol, KafkaProducerProtocolBuilder}

object Predef {


  def kafka_producer(servers: String, acks: String = "1"): KafkaProducerProtocol = KafkaProducerProtocolBuilder(servers, acks)

  def kafka_consumer(servers: String, group: String = "1"): KafkaConsumerProtocol = KafkaConsumerProtocolBuilder(servers, group)

  def producer(name: String): KafkaProductActionBuilder = KafkaProductActionBuilder(name)

  def consumer(name: String): KafkaConsumerActionBuilder = KafkaConsumerActionBuilder(name)
}
